Vương Tông Diễn hiếu tửu sắc, nhạc du hí. Vương Kiến thường trông thấy cảnh Thái tử cùng chư vương chơi đá gà và đánh cầu, rèo hò ầm ĩ; Vương Kiến than thở: "Ta bách chiến mà lập nên cơ nghiệp, lũ này có thể giữ gìn được không?" Vương Kiến do vậy bực bội vơi Trương Cách, song với sự trợ giúp của Từ hiền phi, Trương Cách không bị ông bãi chức tể tướng. Tuy nhiên, Vương Kiến xem xét việc cho Vương Tông Kiệt thay thế ngôi vị Thái tử. Đến khi Vương Tông Kiệt đột ngột qua đời, Vương Kiến nghi ngờ rằng Tông Kiệt bị mưu sát, song sau đó không có thêm hành động nào nhằm thay thế Vương Tông Diễn.[1]

Cũng vào năm 918, Vương Kiến lâm bệnh nặng, và do tin tưởng Bắc diện hành doanh chiêu thảo sứ kiêm Trung thư lệnh Vương Tông Bật (王宗弼) trầm tĩnh hữu mưu nên có thể trợ giúp cho vị hoàng đế trẻ tuổi sau này, Vương Kiến triệu Vương Tông Bật từ chiến trường với Kỳ về nhậm chức Mã bộ đô chỉ huy sứ. Vương Kiến triệu các đại thần đến tẩm điện, nói với họ:[1]

Thái tử nhân nhược, Trẫm không thể làm trái lời thỉnh của chư công nên mới lập làm Thái tử. Nếu như nó không kham nổi đại nghiệp, có thể đưa sang biệt cung, song mong đừng giết chết. Thay vào đó, Chư công hãy chọn ra một tử đệ khác của Vương thị và giúp đỡ. Huynh đệ của Từ phi thì chỉ có thể ban cho họ bổng lộc và tước vị, chớ đừng cho họ cầm binh dự chính, bảo toàn tông tộc của họ.

Tuy nhiên, sau đó Nội phi long sứ Đường Văn Ỷ sai người canh giữ ở cửa cung, muốn trừ khử các đại thần, tự mình kiểm soát Vương Kiến. Tuy nhiên, đồng đảng của Đường Văn Ỷ là Nội hoàng thành sứ Phan Tại (潘在迎) tiết lộ kế hoạch cho các đại thần, các đại thần xông vào cung và đuổi Đường Văn Ỷ. Không lâu sau, Vương Kiến ban di chiếu bổ nhiệm Tống Quang Tự (宋光嗣) làm Nội xu mật sứ, cùng với Vương Tông Bật, Vương Tông Dao, Vương Tông Oản, Vương Tông Quỳ phụ chính cho Vương Tông Diễn rồi qua đời; Vương Tông Diễn kế vị.[1] Ông cải danh thành Vương Diễn.[4]